import unittest
# 导入测试模块
from test01_unittest import Test01
from test02 import Test02
# 创建测试套件
suite = unittest.TestSuite()
# 将测试模块中的测试用例添加到测试套件中
suite.addTest(Test01('test_01'))
suite.addTest(Test01('test_02'))
# 添加测试模块中所有的测试用例
# suite.addTest(unittest.makeSuite(Test02)) # 即将弃用，不建议使用
suite.addTest(unittest.TestLoader().loadTestsFromTestCase(Test02))
# 最后执行测试套件中所有的测试用例
if __name__ == '__main__':
    # 创建测试套件运行器
    runner = unittest.TextTestRunner()
    # 运行测试套件
    runner.run(suite)